#pragma once
#include <iostream>
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <cstring>

using namespace std;

class InetAddr
{
    void getAddress(string *ip, uint16_t *port)
    {
        *ip = inet_ntoa(_addr.sin_addr);
        *port = ntohs(_addr.sin_port);
    }
public:
    InetAddr(const struct sockaddr_in& addr):_addr(addr)
    {
        getAddress(&_ip, &_port);
    }

    bool operator==(const InetAddr &addr)
    {
        if(addr._ip == _ip && addr._port == _port)
        {
            return true;
        }
        return false;
    }

    struct sockaddr_in getAddr()
    {
        return _addr;
    }

    string Ip()
    {
        return _ip;
    }

    uint16_t Port()
    {
        return _port;
    }

    ~InetAddr()
    {}

private:
    struct sockaddr_in _addr;
    uint16_t _port;
    string _ip;
};